from executor import SpanLinkExecutor

if __name__ == '__main__':
    # 'checkout', 'cart'
    SpanLinkExecutor(data_file='data/cbt/checkout_100_user50.json',
                     blocked_services=['frontend-proxy', 'frontend', 'shipping',
                                       'recommendation']).run_pipeline()

    # 'frontend'
    SpanLinkExecutor(data_file='data/cbt/frontend_500_user10.json',
                     blocked_services=['frontend-proxy', 'checkout', 'cart', 'shipping',
                                       'recommendation']).run_pipeline()

    # 'shipping'
    SpanLinkExecutor(data_file='data/cbt/shipping_20_user10.json',
                     blocked_services=['frontend-proxy', 'frontend', 'checkout', 'cart',
                                       'recommendation'], predict_method='FCFS').run_pipeline()

    # 'recommendation'
    SpanLinkExecutor(data_file='data/cbt/frontend_500_user10.json',
                     blocked_services=['frontend-proxy', 'checkout', 'cart', 'shipping',
                                       'frontend']).run_pipeline()
